KarlComSe / SvenskaElsparkcyklarAB

import { InjectRepository } from '@nestjs/typeorm';

import { Repository } from 'typeorm';

import { User } from './entities/user.entity';

import { UpdateUserDto } from './dto/update-user.dto/update-user.dto';

import { AdjustFundsDto } from './dto/update-user.dto/adjust-funds.dto';

@Injectable()

export class UsersService {

  constructor(

    @InjectRepository(User)

    private userRepository: Repository<User>,

  ) {}

  // Find all customers

  async findAll(): Promise<User[]> {

    return this.userRepository.find();

  }

  // Find a customer by ID

  async findById(githubId: string): Promise<User> {

    const user = await this.userRepository.findOne({ where: { githubId } });

    if (!user) {

      throw new NotFoundException('User not found');

    }

    return user;

  }

  // Update customer fields

  async update(githubId: string, updateUserDto: UpdateUserDto): Promise<User> {

    const user = await this.findById(githubId);

    return this.userRepository.save({ ...user, ...updateUserDto });

  }

  async adjustFunds(githubId: string, adjustFundsDto: AdjustFundsDto): Promise<User> {

    const user = await this.userRepository.findOneBy({ githubId });

    if (!user) {

      throw new NotFoundException(`User with GitHub ID ${githubId} not found.`);

    }

    // Update balance and/or isMonthlyPayment only if provided

    if (adjustFundsDto.balance !== undefined) {

      user.balance = adjustFundsDto.balance;

    }

    if (adjustFundsDto.isMonthlyPayment !== undefined) {

      user.isMonthlyPayment = adjustFundsDto.isMonthlyPayment;

    }

    return this.userRepository.save(user);

  }

}